#!/usr/bin/python3
# -*- coding:utf-8 -*-
"""
@File: TestAiPrompt.py
@Author: lijk34925 
@Date: 2025/5/9 9:37
@desc: 请补充本模块功能说明
"""
import pytest

from BladeFerm.Lib.data import load_data
from ollama.Prompt import act_prompt


class TestAiPrompt:

    @pytest.mark.parametrize('input_data,ui_expect', load_data('../data/test_ai_prompt_data.yaml'))
    def test_ollama_prompt(self, input_data, ui_expect):
        prompt = input_data.get("prompt")
        res = act_prompt(prompt, "", [])
        print(f"res: {res}")
